package com.xt.LibaryDeskManager.utils;

import com.xt.LibaryDeskManager.dto.UserMessage;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 学号格式化工具
 * @author 童归权
 * @version 1.0 2021/2/7
 */
public class Trans {
    /**
     * 按照学院分类
     * @param userMessages
     * @return
     */
    public static Map<Integer,List<UserMessage>> xueyuan(List<UserMessage> userMessages){
        Map<Integer,List<UserMessage>> map1=new HashMap<>();
        map1.put(1,new ArrayList<UserMessage>());
        map1.put(2,new ArrayList<UserMessage>());
        map1.put(3,new ArrayList<UserMessage>());
        map1.put(4,new ArrayList<UserMessage>());
        map1.put(5,new ArrayList<UserMessage>());
        map1.put(6,new ArrayList<UserMessage>());
        map1.put(7,new ArrayList<UserMessage>());
        map1.put(8,new ArrayList<UserMessage>());
        map1.put(9,new ArrayList<UserMessage>());
        map1.put(10,new ArrayList<UserMessage>());
        map1.put(11,new ArrayList<UserMessage>());


        for (UserMessage message : userMessages) {
           switch( message.getId().substring(2,4)){
               case "01":
                       map1.get(1).add(message);
                   break;
               case "02":
                       map1.get(2).add(message);
                   break;
               case "03":
                       map1.get(3).add(message);
                   break;
               case "04":
                       map1.get(4).add(message);
                   break;
               case "05":
                       map1.get(5).add(message);
                   break;
               case "06":
                       map1.get(6).add(message);
                   break;
               case "07":
                       map1.get(7).add(message);
                   break;
               case "08":
                       map1.get(8).add(message);
                   break;
               case "09":
                       map1.get(9).add(message);
                   break;
               case "10":
                       map1.get(10).add(message);
                   break;
               case "11":
                       map1.get(11).add(message);
                   break;
           }
        }
        return map1;
    }

    /**
     * 按照年级分类
     * @param userMessages
     * @return
     */
    public static Map<Integer,List<UserMessage>> nianji(List<UserMessage> userMessages){
        Map<Integer,List<UserMessage>> map1=new HashMap<>();
        map1.put(17,new ArrayList<UserMessage>());
        map1.put(18,new ArrayList<UserMessage>());
        map1.put(19,new ArrayList<UserMessage>());
        map1.put(20,new ArrayList<UserMessage>());
        for (UserMessage message : userMessages) {
            switch( message.getId().substring(0,2)){
                case "17":
                    map1.get(17).add(message);
                    break;
                case "18":
                    map1.get(18).add(message);
                    break;
                case "19":
                    map1.get(19).add(message);
                    break;
                case "20":
                    map1.get(20).add(message);
                    break;
            }
        }
        return map1;
    }
}
